#ifndef FACE_FEATURE_ARCFACE_API_H
#define FACE_FEATURE_ARCFACE_API_H

#include <iostream>
#include <memory>
#include <opencv2/opencv.hpp>

namespace arcfacefeature
{

class ArcfaceFeature_api
{

public:

    ArcfaceFeature_api(const int gpu_id, const std::string &onnx_path, const std::string &trt_dir, const int max_batch = 1);

    ~ArcfaceFeature_api();

public:

    std::vector<float >  Execute(const cv::Mat &img);
    
private:

    ArcfaceFeature_api();

    class Impl;

    std::unique_ptr <Impl> impl_;


};








};//namespace arcfacefeature





#endif